<!DOCTYPE HTML>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>My WebSocket</title>
    <style>
        #message{
            margin-top:40px;
            border:1px solid gray;
            padding:20px;
        }
    </style>
</head>
<body>
<input type="text" id="token"/>
<button onclick="conectWebSocket()">连接WebSocket</button>
<button onclick="closeWebSocket()">断开连接</button>
<hr />
<br />
消息：<input id="text" type="text" /><br/>
频道号：<input id="toUser" type="text" /><br/>
上传文件：<input type="file" id="f" onchange="chooseFile()"><br/>
<button onclick="send()">发送消息</button><br/>
----------------------------------<br/>
文字：<div id="message"></div><br/>
----------------------------------<br/>
图片：<div id="main"></div>
</body>
<script type="text/javascript">
    var websocket = null;
    function conectWebSocket(){
        var token = document.getElementById("token").value;
        if(token===""){
            alert("请输入昵称");
            return;
        }

        //判断当前浏览器是否支持WebSocket
        if ('WebSocket'in window) {
            websocket = new WebSocket("ws://localhost/websocket/"+token);
            // websocket = new WebSocket("ws://localhost/websocket");

            // websocket = new WebSocket("ws://localhost/websocket?id=aaa","12345678900");
            // websocket = new WebSocket("ws://localhost/websocket?id=aaa","12345678900");

            // websocket = new WebSocket("ws://localhost/websocket/");
            // websocket = new WebSocket("ws://localhost/websocket/"+nickname);
            // websocket = new WebSocket("ws://localhost/websocket");

        } else {
            alert('Not support websocket')
        }
        //连接发生错误的回调方法
        websocket.onerror = function() {
            setMessageInnerHTML("error");
        };
        //连接成功建立的回调方法
        websocket.onopen = function(event) {
            setMessageInnerHTML("Loc MSG: 成功建立连接");
            setMessageInnerHTML("event:"+event+"    json:"+JSON.stringify(event));
        }

        //接收到消息的回调方法
        websocket.onmessage = function(event) {
            // setMessageInnerHTML(event.data);
            // document.getElementById('message').innerHTML += event.data + '<br/>';
            var innerHTML = event.data;
            // document.getElementById('message').innerHTML += innerHTML + '<br/>';
            //解析json
            var jsonStr = JSON.parse(innerHTML);
            //1为文本消息
            if(jsonStr.type==3){
                document.getElementById('main').innerHTML += "<img style='width:200px;height:200px;' width='150px' src="+jsonStr.msg+">";
                //2为图片消息
            }else{
                document.getElementById('message').innerHTML += innerHTML + '<br/>';
            }
        }
        //连接关闭的回调方法
        websocket.onclose = function() {
            setMessageInnerHTML("Loc MSG:关闭连接");
        }
        //监听窗口关闭事件，当窗口关闭时，主动去关闭websocket连接，防止连接还没断开就关闭窗口，server端会抛异常。
        window.onbeforeunload = function() {
            websocket.close();
        }
    }
    //将消息显示在网页上
    function setMessageInnerHTML(innerHTML) {
        document.getElementById('message').innerHTML += innerHTML + '<br/>';
    }
    //关闭连接
    function closeWebSocket() {
        websocket.close();
    }
    // //发送消息
    // function send() {
    //     var message = document.getElementById('text').value;
    //     websocket.send(message);
    // }


    //发送图片消息
    function chooseFile() {
        var files = document.querySelector("#f").files
        if(files.length>0){
            var fileReader = new FileReader();
            fileReader.readAsDataURL(files[0])
            fileReader.onload=function (e) {
                alert(e.target.result);
                var s = JSON.stringify({type:3,msg:e.target.result});
                websocket.send(s);
            }
        }
    }


    //发送消息
    function send() {
        //获取输入的文本信息进行发送
        var message = document.getElementById('text').value;
        var toUser = document.getElementById('toUser').value;
        var socketMsg = {msg:message,toUser:toUser};
        if(toUser == ''){
            //群聊.
            socketMsg.type = 0;
        }else{
            //单聊.
            socketMsg.type = 1;
        }
        websocket.send(JSON.stringify(socketMsg));
    }

</script>
</html>

